# %matplotlib inline

import cv2
import matplotlib.pyplot as plt
import paddlex as pdx

test_jpg = '/media/string/My' + ' Passport/CULane/CULane' + '/JPEGImages/26630.jpg'
model_ = pdx.load_model('output/deeplabv3p_r50vd/epoch_9')
result_ = model_.predict(test_jpg)
pdx.seg.visualize(test_jpg, result_, weight=0.0, save_dir='./')

img_ = cv2.imread(test_jpg)
cv2.imshow('1234', img_)
cv2.waitKey()


